from enum import unique

from myapp.const.base import TEnum


# Defines an enumeration for different categories of tasks.
@unique
class EnumTaskCategory(TEnum):
    """
    训练、推理资源池类型
    """

    # Represents the 'Auto Learning' task category.
    auto_learning = 1  # 自动学习
    # Represents the 'Finetune' task category.
    finetune = 2  # 微调
    # Represents the 'Post Pretraining' task category.
    post_pretraining = 3  # 增量预训练
    # Represents the 'Model Compression' task category.
    compress = 4  # 模型压缩
    # Represents the 'Model Merging' task category.
    merge = 5  # 模型融合
    # Represents the 'Reinforcement Learning from Human Feedback' (RLHF) task category.
    rlhf = 6 # 偏好对齐